/*
 * Copyright (c) 2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { native } from 'qvog-dsl';
import { Variable, Value, InvokeStmt, Constant } from 'qvog-types';
import { QHint } from '~/adapter/Defines';
import { DataFlow } from '~/lib/flow';

export default native<QHint>()()
    .match((node: Value): node is Value => {
        return node.stream().any(s =>
            s instanceof Constant &&
            s.type.name === "string" &&
            s.value.startsWith("http://")
        )
    })
    .when((node, { native }) => {
        const sinkNodes = native.api.nodes(s => {
            if (!(s instanceof InvokeStmt)) return false;
            const expr = s.expression;
            const isFetch = ["fetch", "request", "query"].includes(expr.target) &&
                expr.argCount > 0 &&
                expr.args[0] instanceof Variable;

            const isAxios = (expr.base instanceof Variable) &&
                (() => {
                    const base = expr.base as Variable;
                    return base.name === "axios";
                })() &&
                ["get", "post", "put", "delete"].includes(expr.target) &&
                expr.argCount > 0 &&
                expr.args[0] instanceof Variable;

            const isXHR = expr.base instanceof Variable &&
                expr.base?.type.name.includes("XMLHttpRequest") &&
                expr.target === "open" &&
                expr.argCount > 1 &&
                expr.args[1] instanceof Variable;
            if (isFetch || isAxios || isXHR)
                return true;
            return false;
        }, 'Sink')
        return native.api.exists(node, '', sinkNodes, "Flow", () => new DataFlow());
    })
    .report({ table: 'Flow' });
